package me.mingshan.leetcode;

import java.util.Arrays;

/**
 * https://leetcode.cn/problems/remove-element/
 * <p>
 * 给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素。元素的顺序可能发生改变。然后返回 nums 中与 val 不同的元素的数量。
 * <p>
 * 假设 nums 中不等于 val 的元素数量为 k，要通过此题，您需要执行以下操作：
 * <p>
 * 更改 nums 数组，使 nums 的前 k 个元素包含不等于 val 的元素。nums 的其余元素和 nums 的大小并不重要。
 * 返回 k。
 * <p>
 * <p>
 * 示例 1：
 * <p>
 * 输入：nums = [3,2,2,3], val = 3
 * 输出：2, nums = [2,2,_,_]
 * 解释：你的函数函数应该返回 k = 2, 并且 nums 中的前两个元素均为 2。
 * 你在返回的 k 个元素之外留下了什么并不重要（因此它们并不计入评测）。
 * 示例 2：
 * <p>
 * 输入：nums = [0,1,2,2,3,0,4,2], val = 2
 * 输出：5, nums = [0,1,4,0,3,_,_,_]
 * 解释：你的函数应该返回 k = 5，并且 nums 中的前五个元素为 0,0,1,3,4。
 * 注意这五个元素可以任意顺序返回。
 * 你在返回的 k 个元素之外留下了什么并不重要（因此它们并不计入评测）。
 *
 * @author hanjuntao
 * @date 2025/6/10 0010
 */
public class L_27_remove_element {

    public static void main(String[] args) {
        int[] source = {3, 2, 2, 3};
        int val = 3;
        System.out.println(removeElement2(source, val));

        int[] source2 = {0, 1, 2, 2, 3, 0, 4, 2};
        int val2 = 2;
        //System.out.println(removeElement(source2, val2));

        int[] source3 = {1};
        int val3 = 1;
        //System.out.println(removeElement(source3, val3));

        int[] source4= {4,5};
        int val4 = 5;
        //System.out.println(removeElement(source4, val4));

        int[] source5= {4,4,0,1,0,2};
        int val5 = 0;
        //System.out.println(removeElement(source4, val4));
    }

    /**
     * 思路，循环检测，当遇到与val相等时，将后面的元素全部移动到前面，然后返回数组长度减去移动的元素个数
     *
     * 注意，当匹配最后一个元素时，直接终止
     *
     * @param nums
     * @param val
     * @return
     */
    public static int removeElement(int[] nums, int val) {
        if (nums == null || nums.length == 0) {
            return 0;
        }

        int count = 0;
        int i = 0;
        while (i < nums.length) {
            if (nums[i] == val) {
                for (int j = i; j < nums.length - 1; j++) {
                    nums[j] = nums[j + 1];
                    nums[j + 1] = Integer.MAX_VALUE;
                }

                if (i == nums.length - 1) {
                    break;
                }
                count++;
            } else {
                i++;
            }
        }

        System.out.println(Arrays.toString(nums));

        return nums.length - count;
    }

    /**
     * 思路，双指针，
     * 所有与val相等的元素，移动到数组的末尾
     *
     *
     * @param nums
     * @param val
     * @return
     */
    public static int removeElement2(int[] nums, int val) {
        // 指向下一个插入的位置
        int left = 0;
        // 指向下一个处理的元素的位置
        int right = 0;

        for (right = 0; right < nums.length; right++) {
            // 不等于val，需要将当前元素插入到left位置
            // 同时left指针后移
            if (nums[right] != val) {
                nums[left] = nums[right];
                left++;
            }
        }
        return left;
    }
}
